# 处理土地分类数据
import pandas as pd
import numpy as np

xlsfile = 'data/土地利用分类数据/程序处理/滇中县域土地利用分类面积汇总2020年30米.xlsx'
sheet1 = '生态价值赋值表'
sheet2 = '土地分类面积汇总'

# 加载土地分类面积汇总
dataframe_value = pd.read_excel(xlsfile, sheet_name=sheet1)
dataframe_data = pd.read_excel(xlsfile, sheet_name=sheet2)
print('-----------生态价值赋值表-----------')
print(dataframe_value)
print('-----------原始土地分类面积汇总-----------')
print(dataframe_data)
index_xzq = np.array(dataframe_data['行政区'])
print('-----------行政区-----------')
print(index_xzq)
dataframe_data.drop(columns="行政区", axis=1, inplace=True)
dataframe_data.set_index([index_xzq], inplace=True)

scores_df = pd.DataFrame(columns=dataframe_data.columns, index=dataframe_data.index)
for col in dataframe_data.columns:
    scores_df[col] = dataframe_data[col] * dataframe_value[col][0]
scores_df['综合指数'] = scores_df.sum(axis=1)
print('-----------土地分类贡献指数-----------')
print(scores_df)
# 保存为excel
scores_df.to_excel('生态环境指数2020.xlsx')

dataframe_data['总面积'] = dataframe_data.sum(axis=1)
print('-----------土地分类面积汇总-----------')
print(dataframe_data)
# 保存为excel
dataframe_data.to_excel('土地分类面积汇总2020.xlsx')

result_df = pd.DataFrame(columns=['综合指数', '总面积', '生态水平'], index=dataframe_data.index)
result_df['综合指数'] = scores_df['综合指数']
result_df['总面积'] = dataframe_data['总面积']
result_df['生态水平'] = result_df['综合指数'] / result_df['总面积']
print('-----------生态水平汇总-----------')
print(result_df)
# 保存为excel
result_df.to_excel('生态水平汇总2020.xlsx')




